class Config:
    # train process
    EPOCHS = 100
    BATCH = 16

    # dataset
    width, height = 320, 320
    train_ratio = 0.9